Embabel Agent Documentation/com.embabel.agent.rag.neo.ogm

Package-level declarations

Types

class NeoOgmKnowledgeGraphService(sessionFactory: <Error class: unknown class>, ogmCypherSearch: OgmCypherSearch, modelProvider: <Error class: unknown class>, properties: NeoOgmKnowledgeGraphServiceProperties) : SchemaResolver, ChunkRepository

Implements several interfaces to read and write knowledge graph data to Neo4j using Neo4j OGM.

NeoOgmKnowledgeGraphServiceProperties

data class NeoOgmKnowledgeGraphServiceProperties(val chunkNodeName: String = "Document", val entityNodeName: String = "Entity")

OgmCypherSearch

class OgmCypherSearch(sessionFactory: <Error class: unknown class>, queryResolver: LogicalQueryResolver) : CypherSearch

OgmMappedEntity

abstract class OgmMappedEntity(val id: String) : MappedEntity

Superclass for all entities that are mapped using Neo4j OGM.

OgmMappedNamedEntity

abstract class OgmMappedNamedEntity(val name: String, val id: String) : OgmMappedEntity

OgmRagService

class OgmRagService(modelProvider: <Error class: unknown class>, queryRunner: OgmCypherSearch, schemaResolver: SchemaResolver, platformTransactionManager: <Error class: unknown class>, properties: OgmRagServiceProperties = OgmRagServiceProperties()) : RagService

Performs RAG queries in readonly transactions using Neo4j OGM. Requires a Neo4j OGM PlatformTransactionManager to be configured in the Spring context.

OgmRagServiceProperties

data class OgmRagServiceProperties(val vectorIndex: String = "spring-ai-document-index")

SchemaDrivenCypherRagQueryGenerator

class SchemaDrivenCypherRagQueryGenerator(modelProvider: <Error class: unknown class>, schema: KnowledgeGraphSchema) : CypherRagQueryGenerator

Generate RAG queries from a given schema

Functions

cleanLlmCypher

fun cleanLlmCypher(rawResponse: String): String

Try to extract Cypher from the response